package club.xiaojiawei.strings;

import java.util.HashMap;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 5/4/22 12:51 AM
 * @question 2024. 考试的最大困扰度
 * @description 一位老师正在出一场由 n 道判断题构成的考试，每道题的答案为 true （用 'T' 表示）或者 false （用 'F' 表示）。老师想增加学生对自己做出答案的不确定性，方法是 最大化 有 连续相同 结果的题数。（也就是连续出现 true 或者连续出现 false）。
 * 给你一个字符串 answerKey ，其中 answerKey[i] 是第 i 个问题的正确结果。除此以外，还给你一个整数 k ，表示你能进行以下操作的最多次数：
 * 每次操作中，将问题的正确答案改为 'T' 或者 'F' （也就是将 answerKey[i] 改为 'T' 或者 'F' ）。
 * 请你返回在不超过 k 次操作的情况下，最大 连续 'T' 或者 'F' 的数目。
 */
public class MaxConsecutiveAnswers2024 {

    public static void main(String[] args) {
        String answerKey = "TTTTTTTTTFFFFFTT";
        int result = maxConsecutiveAnswers3(answerKey, 3);
        System.out.println(result);
    }

    /**
     * 双指针-HashMap
     * @param answerKey
     * @param k
     * @return
     */
    public static int maxConsecutiveAnswers(String answerKey, int k) {
        int l = 0,r = 0, max = 0;
        var map = new HashMap<Character, Integer>(2);
        char[] chars = answerKey.toCharArray();
        while (r < answerKey.length()){
            map.put(chars[r], map.getOrDefault(chars[r], 0) + 1);
            max = Math.max(max, map.get(chars[r]));
            r++;
            if (r - l - max > k){
                map.put(chars[l], map.get(chars[l]) - 1);
                l++;
            }
        }
        return r - l;
    }

    /**
     * 双指针-int[]
     * @param answerKey
     * @param k
     * @return
     */
    public static int maxConsecutiveAnswers2(String answerKey, int k) {
        if (answerKey.length() <= k + 1){
            return answerKey.length();
        }
        int l = 0,r = 0, max = 0;
        int[] ints = new int[2];
        char[] chars = answerKey.toCharArray();
        while (r < answerKey.length()){
            max = Math.max(max, ++ints[chars[r] == 'T'? 0 : 1]);
            r++;
            if (r - l - max > k){
                ints[chars[l] == 'T'? 0 : 1]--;
                l++;
            }
        }
        return r - l;
    }

    /**
     * 民间-双指针（优化到极致）
     * @param answerKey
     * @param k
     * @return
     */
    public static int maxConsecutiveAnswers3(String answerKey, int k) {
        byte[] answers = answerKey.getBytes();
        int[] save = new int[128];
        int start = 0;
//        foreach隐藏了一个指针
        for (byte answer : answers) {
            save[answer]++;
            if (Math.min(save['T'], save['F']) > k) {
                save[answers[start++]]--;
            }
        }
        return answers.length - start;
    }
}
